#include "agx_kit.h"

void AGXKit::setModel(const std::string& model) {
    this->model = model;
}

void AGXKit::setAI(const std::string& ai) {
    this->ai = ai;
}

void AGXKit::setCudaCores(int cuda_cores) {
    this->cuda_cores = cuda_cores;
}

void AGXKit::setTensorCores(int tensor_cores) {
    this->tensor_cores = tensor_cores;
}

void AGXKit::setVRAM(const std::string& vram) {
    this->vram = vram;
}

void AGXKit::setStorage(const std::string& storage) {
    this->storage = storage;
}

void AGXKit::print() const {
    std::cout << "AGX套件型号: " << model << std::endl;
    std::cout << "AI算力: " << ai << std::endl;
    std::cout << "CUDA核心数: " << cuda_cores << std::endl;
    std::cout << "Tensor核心数: " << tensor_cores << std::endl;
    std::cout << "显存: " << vram << std::endl;
    std::cout << "存储: " << storage << std::endl;
}

void AGXKit::save(std::ofstream& out) const {
    out << "AGX套件型号: " << model << std::endl;
    out << "AI算力: " << ai << std::endl;
    out << "CUDA核心数: " << cuda_cores << std::endl;
    out << "Tensor核心数: " << tensor_cores << std::endl;
    out << "显存: " << vram << std::endl;
    out << "存储: " << storage << std::endl;
}